package com.example.mq.mqserver.datacenter;

import com.example.mq.MqApplication;
import com.example.mq.mqserver.core.Binding;
import com.example.mq.mqserver.core.Exchange;
import com.example.mq.mqserver.core.ExchangeType;
import com.example.mq.mqserver.core.MQueue;
import com.example.mq.mqserver.mapper.MetaMapper;
import lombok.extern.slf4j.Slf4j;

import java.io.File;
import java.util.List;

/**
 * 这个类给上层提供数据库相关的操作
 * 初始化数据库 (创建目录，创建表)
 * 交换机、队列、绑定的增删改查
 */
@Slf4j
public class DataBaseManager {

    private MetaMapper metaMapper;
    /**
     * 针对数据库进行初始化操作
     * 建库 + 建表 + 插入默认的数据
     * 数据库文件不存在的情况下才执行
     */
    public void init() {
        // 通过 getBean 的方式获取
        metaMapper = MqApplication.context.getBean(MetaMapper.class);
        if (!checkDBExists()) {
            // 如果数据库文件不存在，就建表，插入默认数据（首次执行数据库操作时，会自动创建数据库文件）
            // 项目配置的数据库文件路径：./data/meta.db
            // 需要保证 data 目录存在
            File dataDir = new File("./data");
            dataDir.mkdirs();
            // 建表 + 插入默认数据
            createTable();
            createDefaultData();
            log.info("数据库初始化完成！");
        } else {
            log.info("数据库已经存在！");
        }
    }

    /**
     * 检查数据库文件是否存在
     * @return 数据库存在返回true
     */
    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        if (file.exists()) {
            return true;
        }
        return false;
    }

    /**
     * 建表
     */
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        log.info("创建表完成！");
    }

    /**
     * 插入默认数据
     */
    private void createDefaultData() {
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        log.info("插入初始数据完成！");
    }

    /**
     * 删除数据库（删除数据库文件和目录）
     */
    public void deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if (ret) {
            log.info("删除数据库文件成功！");
        } else {
            log.info("删除数据库文件失败！");
        }
        File dataDir = new File("./data");
        ret = dataDir.delete();		// detele 删除目录时，想要保证目录是空的
        if (ret) {
            log.info("删除数据库目录成功！");
        } else {
            log.info("删除数据库目录失败！");
        }
    }

    // 封装交换机、队列、绑定的插入、删除、查找操作
    public void insertExchange(Exchange exchange) {
        metaMapper.insertExchange(exchange);
    }

    public List<Exchange> selectAllExchanges() {
        return metaMapper.selectAllExchanges();
    }

    public void deleteExchange(String exchangeName) {
        metaMapper.deleteExchange(exchangeName);
    }

    public void insertQueue(MQueue queue) {
        metaMapper.insertQueue(queue);
    }

    public List<MQueue> selectAllQueues() {
        return metaMapper.selectAllQueues();
    }

    public void deleteQueue(String queueName) {
        metaMapper.deleteQueue(queueName);
    }

    public void insertBinding(Binding binding) {
        metaMapper.insertBinding(binding);
    }

    public List<Binding> selectAllBindings() {
        return metaMapper.selectAllBindings();
    }

    public void deleteBinding(Binding binding) {
        metaMapper.deleteBinding(binding);
    }
}
